from selenium import webdriver


def open_browser(browser='chrome'):
    """
    打开浏览器,返回driver
    设置浏览器类,默认谷歌浏览器
    """

    # 打开浏览器,设置driver
    if browser == 'chrome':
        return webdriver.Chrome()
    elif browser == 'firefox':
        return webdriver.Firefox()
    elif browser == 'ie':
        return webdriver.Ie()
    elif browser == 'edge':
        return webdriver.Edge()
    else:
        print('不支持该浏览器')